package com.zzyl.nursing.service.impl;

import java.time.LocalDateTime;
import java.time.LocalTime;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoUnit;
import java.util.*;
import java.util.concurrent.TimeUnit;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.date.LocalDateTimeUtil;
import cn.hutool.core.text.CharSequenceUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.zzyl.common.constant.CacheConstants;
import com.zzyl.common.utils.DateUtils;
import com.zzyl.nursing.config.WebSocketServer;
import com.zzyl.nursing.domain.AlertData;
import com.zzyl.nursing.domain.DeviceData;
import com.zzyl.nursing.mapper.DeviceMapper;
import com.zzyl.nursing.service.IAlertDataService;
import com.zzyl.nursing.vo.AlertNotifyVo;
import com.zzyl.system.mapper.SysUserRoleMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import com.zzyl.nursing.mapper.AlertRuleMapper;
import com.zzyl.nursing.domain.AlertRule;
import com.zzyl.nursing.service.IAlertRuleService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

/**
 * 报警规则Service业务层处理
 * 
 * @author alexis
 * @date 2025-01-15
 */
@Service
public class AlertRuleServiceImpl extends ServiceImpl<AlertRuleMapper, AlertRule> implements IAlertRuleService
{
    @Autowired
    private AlertRuleMapper alertRuleMapper;

    @Autowired
    private RedisTemplate<Object, Object> redisTemplate;

    @Autowired
    private IAlertDataService alertDataService;

    @Autowired
    private DeviceMapper deviceMapper;

    @Autowired
    private SysUserRoleMapper sysUserRoleMapper;

    @Autowired
    private WebSocketServer webSocketServer;

    /**
     * 查询报警规则
     * 
     * @param id 报警规则主键
     * @return 报警规则
     */
    @Override
    public AlertRule selectAlertRuleById(Long id)
    {
        return alertRuleMapper.selectById(id);
    }

    /**
     * 查询报警规则列表
     * 
     * @param alertRule 报警规则
     * @return 报警规则
     */
    @Override
    public List<AlertRule> selectAlertRuleList(AlertRule alertRule)
    {
        return alertRuleMapper.selectAlertRuleList(alertRule);
    }

    /**
     * 新增报警规则
     * 
     * @param alertRule 报警规则
     * @return 结果
     */
    @Override
    public int insertAlertRule(AlertRule alertRule)
    {
        return alertRuleMapper.insert(alertRule);
    }

    /**
     * 修改报警规则
     * 
     * @param alertRule 报警规则
     * @return 结果
     */
    @Override
    public int updateAlertRule(AlertRule alertRule)
    {
        return alertRuleMapper.updateById(alertRule);
    }

    /**
     * 批量删除报警规则
     * 
     * @param ids 需要删除的报警规则主键
     * @return 结果
     */
    @Override
    public int deleteAlertRuleByIds(Long[] ids)
    {
        return alertRuleMapper.deleteBatchIds(Arrays.asList(ids));
    }

    /**
     * 删除报警规则信息
     * 
     * @param id 报警规则主键
     * @return 结果
     */
    @Override
    public int deleteAlertRuleById(Long id)
    {
        return alertRuleMapper.deleteById(id);
    }

    /**
     * 报警规则过滤
     */
    @Override
    public void alertFilter() {
        // 1.查询有没有有效的报警规则
        long count = count(Wrappers.<AlertRule>lambdaQuery().eq(AlertRule::getStatus, 1));

        // 2.如果没有有效的规则，直接结束方法
        if (count == 0) {
            return;
        }

        // 3.如果有规则，从Redis中查询所有设备上报的最新的数据 Object：本身的真实类型：List<DeviceData>
        List<Object> deviceDataLists = redisTemplate.opsForHash().values(CacheConstants.IOT_DEVICE_DATA_LATEST);

        // 4.判断数据是否为空，如果为空，结束方法
        if (CollUtil.isEmpty(deviceDataLists)) {
            return;
        }

        // 5.如果数据不为空，遍历数据，将数据一条一条地交给数据过滤报警器方法，判断数据是否满足报警规则
        deviceDataLists.forEach(deviceData -> {
            List<DeviceData> deviceDataList = (List<DeviceData>) deviceData;
            // 再遍历这每一个设备的最新数据，拿到其中一个物模型的数据，交给数据报警处理器方法处理
            deviceDataList.forEach(data -> {
                alertFilter(data);
            });
        });
    }

    /**
     * 接收一个物模型的一条数据，判断是否满足报警规则
     * @param data  一个物模型的一条数据
     */
    private void alertFilter(DeviceData data) {
        // 1.查询所有报警规则，一种是针对该产品的所有规则，另一种是针对某一个设备的独有规则
        List<AlertRule> alertRules = list(Wrappers.<AlertRule>lambdaQuery()
                .eq(AlertRule::getProductKey, data.getProductKey())
                .eq(AlertRule::getFunctionId, data.getFunctionId())
                .eq(AlertRule::getStatus, 1)
                .in(AlertRule::getIotId, -1, data.getIotId()));

        // 2.判断规则是否为空，如果规则为空，结束方法，换下一条数据
        if (CollUtil.isEmpty(alertRules)) {
            return;
        }

        // 3.如果规则不为空，遍历规则，拿到每一个规则，判断数据是否满足规则，如果满足，则发送报警信息
        alertRules.forEach(alertRule -> {
            // 1.判断数据是否在生效时间内
            // 获取到数据的上报时间和报警规则的生效时间，再比对
            LocalTime alarmTime = data.getAlarmTime().toLocalTime();

            // 判断如果数据上报时间距离当前时间超过了1分钟，直接return跳过
            LocalTime now = LocalTime.now();
            long between = ChronoUnit.SECONDS.between(now, alarmTime);
            if (between > 60) {
                return;
            }

            String[] periods = alertRule.getAlertEffectivePeriod().split("~");
            LocalTime begin = LocalTime.parse(periods[0], DateTimeFormatter.ofPattern("HH:mm:ss"));
            LocalTime end = LocalTime.parse(periods[1], DateTimeFormatter.ofPattern("HH:mm:ss"));

            if (alarmTime.isBefore(begin) || alarmTime.isAfter(end)) {
                return;
            }

            String durationKey = "iot:duration:" + alertRule.getProductKey() + ":" + alertRule.getIotId() + ":" + alertRule.getFunctionId() + ":" + alertRule.getId();
            String silenceKey = "iot:silence:" + alertRule.getProductKey() + ":" + alertRule.getIotId() + ":" + alertRule.getFunctionId() + ":" + alertRule.getId();

            // 2.判断数据是否达到了阈值
            // 获取到数据值和阈值和符号（>= <），再比对
            Double dataValue = Double.valueOf(data.getDataValue());
            Double ruleValue = alertRule.getValue();
            int compare = Double.compare(dataValue, ruleValue); // 如果dataValue大于后面的数字，返回1，小于返回-1，等于返回0

            if ("<".equals(alertRule.getOperator()) && compare >= 0 || ">=".equals(alertRule.getOperator()) && compare < 0) {
                // 数据没有达到阈值，删除Redis中的报警数
                redisTemplate.delete(durationKey);
                return;
            }

            // 3.判断是否是否在沉默周期内
            // 从Redis中获取沉默周期，如果沉默周期不为空，直接return跳过
            Object silencePeriod = redisTemplate.opsForValue().get(silenceKey);
            if (ObjectUtil.isNotEmpty(silencePeriod)) {
                return;
            }

            // 4.判断数据是否达到了持续周期
            Integer duration = (Integer) redisTemplate.opsForValue().get(durationKey);

            // 将已报警次数先加1
            if (ObjectUtil.isEmpty(duration)) {
                duration = 1;
            } else {
                duration += 1;
            }

            // 判断加1后的次数是否达到了持续周期
            if (!duration.equals(alertRule.getDuration())) {
                redisTemplate.opsForValue().set(durationKey, duration);
                return;
            }

            // 5.达到了持续周期，数据校验通过，可以保存到alertData表中了
            // 删除Redis中的报警数
            redisTemplate.delete(durationKey);

            // 添加Redis中的沉默周期
            redisTemplate.opsForValue().set(silenceKey, 1, alertRule.getAlertSilentPeriod(), TimeUnit.MINUTES);

            // 批量保存报警数据到数据库中
            batchSaveAlertData(data, alertRule);
        });
    }

    /**
     * 批量保存报警数据到数据库中
     * @param data  触发了报警的数据
     * @param alertRule 触发了报警的规则
     */
    private void batchSaveAlertData(DeviceData data, AlertRule alertRule) {
        List<AlertData> alertDataList = new ArrayList<>();

        // 查询需要报警的通知人
        List<Long> userIds;
        if (alertRule.getAlertDataType() == 0) {
            // 老人异常数据
            // 判断是随身设备还是固定设备
            if (data.getLocationType() == 0) {
                // 随身设备
                userIds = deviceMapper.selectNursingIdsByIotIdWithElder(data.getIotId());
            } else {
                // 固定设备
                userIds = deviceMapper.selectNursingIdsByIotIdWithBed(data.getIotId());
            }
        } else {
            // 设备异常数据
            userIds = sysUserRoleMapper.selectUserIdByRoleName("维修工");
        }

        // 查询管理员的id
        List<Long> managerUserIds = sysUserRoleMapper.selectUserIdByRoleName("超级管理员");

        // 两部分ids合并到第一个集合中
        userIds.addAll(managerUserIds);

        // 遍历要通知的id列表，构造多条报警数据
        userIds.forEach(userId -> {
            // 先将data拷贝成一个新的alertData对象
            AlertData alertData = BeanUtil.toBean(data, AlertData.class);
            alertData.setId(null);
            alertData.setUserId(userId);
            alertData.setAlertRuleId(alertRule.getId());
            String alertReason = CharSequenceUtil.format("{}{}{},持续{}个周期就报警", alertRule.getFunctionName(), alertRule.getOperator(), alertRule.getValue(), alertRule.getDuration());
            alertData.setAlertReason(alertReason);
            alertData.setType(alertRule.getAlertDataType());
            alertData.setStatus(0); // 将状态设为待处理
            alertDataList.add(alertData);
        });

        alertDataService.saveBatch(alertDataList);

        // 通过WebSocket推送消息给前端
        sendAlertMessage2Client(data, alertRule, userIds);
    }

    /**
     * 通过WebSocket推送消息给前端
     * @param data  设备报警数据
     * @param alertRule 报警规则
     * @param userIds   需要推送的用户id
     */
    private void sendAlertMessage2Client(DeviceData data, AlertRule alertRule, List<Long> userIds) {
        AlertNotifyVo alertNotifyVo = BeanUtil.toBean(data, AlertNotifyVo.class);
        alertNotifyVo.setFunctionName(alertRule.getFunctionName());
        alertNotifyVo.setAlertDataType(alertRule.getAlertDataType());
        alertNotifyVo.setNotifyType(1);
        alertNotifyVo.setIsAllConsumer(false);
        // 调用方法，给前端用户推送数据
        webSocketServer.sendMessageToConsumer(alertNotifyVo, userIds);
    }
}